#!/usr/bin/env python
# -*- coding: utf-8 -*-
###########################################################################
#   Copyright (C) 2008 by Serj Poltavsky                                  #
#   serge.uliss at gmail.com                                              #
#                                                                         #
#   This program is free software; you can redistribute it and/or modify  #
#   it under the terms of the GNU General Public License as published by  #
#   the Free Software Foundation; either version 3 of the License, or     #
#   (at your option) any later version.                                   #
#                                                                         #
#   This program is distributed in the hope that it will be useful,       #
#   but WITHOUT ANY WARRANTY; without even the implied warranty of        #
#   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the         #
#   GNU General Public License for more details.                          #
#                                                                         #
#   You should have received a copy of the GNU General Public License     #
#   along with this program. If not, see <http://www.gnu.org/licenses/>   #
###########################################################################

import unittest
from xml.dom import minidom
from dvdauthor import PGC, VOB

class TestPgc(unittest.TestCase):
    def setUp(self):
        self.pgc = PGC()

    def test_init(self):
        self.assertEqual(self.pgc.nodename, 'pgc')
        self.assertEqual(self.pgc.pause, '0')

    def test_append(self):
        self.assertRaises(TypeError, self.pgc.append, 'foo')

    def test_update(self):
        dom = minidom.Document()
        self.pgc.pause = 100
        self.pgc.pre_action = 'pre_action'
        self.pgc.append(VOB('/dev/null'))
        self.pgc.post_action = 'post_action'
        pgc = self.pgc.update(dom, dom)
        self.assertEqual(pgc.getAttribute('pause'), '100')
        self.assertEqual(pgc.nodeName, 'pgc')
        vobs = dom.getElementsByTagName('vob')
        self.assertEqual(len(vobs), 1)
        pre = dom.getElementsByTagName('pre')
        self.assertEqual(len(pre), 1)
        post = dom.getElementsByTagName('post')
        self.assertEqual(len(post), 1)